python week 4 gaded assignment solution